package com.tianji.aigc.config;

import com.alibaba.cloud.ai.dashscope.chat.DashScopeChatModel;
import com.tianji.aigc.constants.SystemConstants;
import com.tianji.aigc.memory.RedisChatMemory;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * SpringAI配置类
 *
 * @Author mr.xing
 * @Date 2025/9/2 20:29
 */
@Configuration   // spring框架注解,表明这是一个配置类,用来代替Bean,代替XML配置文件
public class SpringAiConfig {

//    @Bean  // 表示该方法将返回一个对象, 该对象被spring容器作为Bean进行管理
//    public ChatClient chatClient(DashScopeChatModel chatModel){
//        // 使用ChatClient.builder方法构建一个ChatClient实例,传入chatModel参数代表使用的对话模型
//        return ChatClient.builder(chatModel)
//                // 调用defaultAdvisors方法为ChatClient添加默认顾问,这里是用于记录日志
//                .defaultAdvisors(new SimpleLoggerAdvisor())
//                .build(); // 用来完成chatclient对象的构建并返回
//    }

    @Bean
    public ChatClient chatClient(DashScopeChatModel chatModel, MessageChatMemoryAdvisor chatMemoryAdvisor) {
        return ChatClient
                .builder(chatModel) //对话模型对象
                .defaultSystem(SystemConstants.SYSTEM)//设置默认系统提示词
                .defaultAdvisors(new SimpleLoggerAdvisor()) //日志记录器
                .build();
    }
    @Bean
    public MessageChatMemoryAdvisor chatMemoryAdvisor(RedisChatMemory chatMemory) {
        return new MessageChatMemoryAdvisor(chatMemory);
    }
}
